# config.py
import os
from dotenv import load_dotenv

load_dotenv()

class Config:
    # 数据库配置 (保持不变)
    DB_CONNECTION = os.getenv("DB_CONNECTION", "mysql")
    DB_HOST = os.getenv("DB_HOST", "127.0.0.1")
    DB_PORT = int(os.getenv("DB_PORT", "3306"))
    DB_DATABASE = os.getenv("DB_DATABASE", "name")
    DB_USERNAME = os.getenv("DB_USERNAME", "root")
    DB_PASSWORD = os.getenv("DB_PASSWORD", "rootroot")

    # 豆包模型配置 (只保留豆包)
    DOUBAO_BASE_URI = os.getenv("DOUBAO_BASE_URI", "https://ark.cn-beijing.volces.com/api/v3")
    DOUBAO_GENERATE_MODEL = os.getenv("DOUBAO_GENERATE_MODEL", "ep-20250205161538-s5fzd")  # 用于 generate
    DOUBAO_COMPLETE_MODEL = os.getenv("DOUBAO_COMPLETE_MODEL", "ep-20250226160246-l4mb6") # 用于 complete
    DOUBAO_ANALYZE_ELEMENTS_MODEL = os.getenv("DOUBAO_ANALYZE_ELEMENTS_MODEL", "ep-20250521165119-hgcsq")
    DOUBAO_API_KEY = os.getenv("DOUBAO_API_KEY", "62d49b5e-fe99-43d4-bb01-46790c1a1e20") # 统一的 API 密钥
    DOUBAO_NAME_LENGTHS = [2, 3]

    # 通用配置 (保持不变)
    MAX_CONCURRENCY = 2000
    BATCH_SIZE = 20
    MAX_RETRIES = 5
    AVAILABLE_STYLES = ["传统", "现代", "自然", "古典", "神话", "动物", "色彩", "数字", "国际", "诗意"]